# -*- coding:utf-8 -*-
# 以面向对象的方式改写存钱取钱函数
class Money():
    def __init__(self):
        # 钱包里的钱
        self.my_money = 100
        # 银行卡里的钱
        self.atm_money = 0

    # 存钱
    def save_money(self):
        num = int(input("请输入你要存款的金额："))
        if num <= self.my_money:
            self.atm_money += num
            self.my_money -= num
            print(f"存款成功\n当前银行卡余额为：{self.atm_money}\n钱包的余额为：{self.my_money}")
        else:
            print("钱包余额不足")

    # 取钱
    def get_money(self):
        num = int(input("请输入你要取款的金额："))
        if num <= self.atm_money:
            self.atm_money -= num
            self.my_money -= num
            print(f"取款成功\n当前银行卡余额为：{self.atm_money}\n钱包的余额为：{self.my_money}")
        else:
            print("银行卡余额不足")

    # 主要流程
    def main(self):
        while True:
            function = int(input("输入数字选择对应的功能: 1、存钱  2、取钱  3、退出: "))
            if function == 1:
                self.save_money()
            elif function == 2:
                self.get_money()
            elif function == 3:
                print("欢迎下次使用~~~")
                break
            else:
                print("请重新选择")


# 程序入口
if __name__ == '__main__':
    money = Money()
    money.main()
